from llama_index.core.base.query_pipeline.query import ChainableMixin
from llama_index.embeddings.openai import OpenAIEmbedding
from llama_index.embeddings.ollama import OllamaEmbedding
from llama_index.core import Settings
from llama_index.core.llms import CustomLLM

embed_model = OllamaEmbedding(
  model_name='milkey/dmeta-embedding-zh:f16',
  prompt='Llamas are members of the camelid family',
)
# embed_model = OpenAIEmbedding()
embeddings = embed_model.get_text_embedding('中国的首都是北京')
print(embeddings)

